from diffusers import DiffusionPipeline
import torch
from datetime import datetime

pipe = DiffusionPipeline.from_pretrained(
    "playgroundai/playground-v2.5-1024px-aesthetic",
    local_files_only=True,
    torch_dtype=torch.float16,
    variant="fp16",
).to("cuda")

# prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
# prompt = "Alien in the jungle, cool tones, pastel colors, detailed, 8k"
prompt = "Humanoid rabbit in the jungle, cyberpunk style, cool tones, pastel colors, detailed, 8k"
image = pipe(prompt=prompt, num_inference_steps=50, guidance_scale=3).images[0]

ts = datetime.now().strftime("%Y%m%d%H%M%S")
filename = f"{ts}.jpg"
image.save(filename)